from .Dataset import Dataset
class Partial_iLIDS_TestSet(Dataset):
    def __init__(self, im_dir=None, im_names=None, marks=None, extract_feat_func=None,
                 separate_camera_set=None, single_gallery_shot=None, first_match_break=None, **kwargs):
        super(Partial_iLIDS_TestSet, self).__init__(dataset_size=len(im_names),**kwargs)
        self.im_dir = im_dir
        self.im_names = im_names
        self.marks = marks
        self.extract_feat_func = extract_feat_func
        self.separate_camera_set = separate_camera_set
        self.single_gallery_shot = single_gallery_shot
        self.first_match_break = first_match_break